package cn.dansj.common.mail.engine;

import org.thymeleaf.TemplateEngine;
import org.thymeleaf.dialect.IDialect;
import org.thymeleaf.templatemode.TemplateMode;
import org.thymeleaf.templateresolver.ClassLoaderTemplateResolver;

import java.nio.charset.StandardCharsets;

public class CustomTemplateEngine extends TemplateEngine {
    public CustomTemplateEngine() {
        // 配置模板解析器
        ClassLoaderTemplateResolver templateResolver = new ClassLoaderTemplateResolver();
        // 模板文件路径（相对于类路径）
        templateResolver.setPrefix("/templates/");
        // 模板文件后缀
        templateResolver.setSuffix(".html");
        templateResolver.setTemplateMode(TemplateMode.HTML);
        templateResolver.setCharacterEncoding(StandardCharsets.UTF_8.name());
        // 开发环境关闭缓存
        templateResolver.setCacheable(true);
        // 设置模板解析器
        setTemplateResolver(templateResolver);
    }

    // 支持手动添加自定义方言
    public void addCustomDialect(IDialect dialect) {
        super.addDialect(dialect);
    }
}
